package com.bjpowernode.test.utils;

import java.io.InputStream;
import java.sql.*;
import java.util.Properties;

/*
  工具类：
      要提供两个方法，一个返回链接的方法，一个关闭链接的方法
 */
public class JDBCUtils {

    private static String driver;
    private static String url;
    private static String user;
    private static String pass;

    // 这四个量要进行初始化
    // 静态代码块 注册驱动
    static {

        try{
            // 1、读取资源文件，获得账号密码
            InputStream is = JDBCUtils.class.getClassLoader().getResourceAsStream("db.properties");
            // 创建属性集对象 来读取资源文件中的数据
            Properties properties = new Properties();
            properties.load(is);

            // 2、初始化属性
            driver = properties.getProperty("driver");  //让这些值都等于了 资源文件中的数据
            url = properties.getProperty("url");
            user = properties.getProperty("user");
            pass = properties.getProperty("pass");

            // 3、加载驱动
            Class.forName(driver);

        }catch (Exception e){
            e.printStackTrace();
        }
    }

    public static Connection getConnection() {

        Connection connection = null; //出现异常就会返回一个空值
        try {
            connection = DriverManager.getConnection(url,user,pass);
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return connection;
    }

    public static void close(Connection conn, Statement sta, ResultSet rs){
        //关闭 释放资源
        if(rs != null){
            try {
                rs.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
        if(sta != null){
            try {
                sta.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
        if(conn != null){
            try {
                conn.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
    }
}
     
